package tools

import (
	"fmt"

	"github.com/spf13/viper"
)

type ConfigModel struct {
	Host   string `mapstructure:"host"`
	Server string `mapstructure:"server"`
	Dsn    string `mapstructure:"dsn"`
}

var Config ConfigModel

// 初始化配置文件
func InitConfig() {
	viper.SetConfigName("application.yaml") // 设置配置文件名（不包含扩展名）
	viper.SetConfigType("yaml")             // 设置配置文件类型
	viper.AddConfigPath("./")
	if err := viper.ReadInConfig(); err != nil {
		panic(fmt.Errorf("无法读取配置文件: %s", err))
	}
	if err := viper.Unmarshal(&Config); err != nil {
		panic(fmt.Errorf("无法读取配置文件: %s", err))
	}
}
